export const transTree = (jsonData) => {
    //result存储json数据转为树形结构后的结果。
    //temp为临时对象，将json数据按照id值排序.
    //len是json长度，用于循环遍历结束的条件
    var result = [],
        temp = {},
        len = jsonData.length;
    for (var i = 0; i < len; i++) {
        // 以id作为索引存储元素，可以无需遍历直接快速定位元素
        temp[jsonData[i]["id"]] = jsonData[i];
    }
    for (var j = 0; j < len; j++) {
        var currentElement = jsonData[j];
        // 临时变量里面的当前元素的父元素，即pid的值，与找对应id值
        var tempCurrentElementParent = temp[currentElement["parentId"]];
        // 如果存在父元素，即如果有pid属性
        if (tempCurrentElementParent) {
            // 如果父元素没有chindren键
            if (!tempCurrentElementParent["children"]) {
                // 设上父元素的children键
                tempCurrentElementParent["children"] = [];
            }
            // 给父元素加上当前元素作为子元素
            tempCurrentElementParent["children"].push(currentElement);
        }
        // 不存在父元素，意味着当前元素是一级元素
        else {
            result.push(currentElement);
        }
    }
    return result;
};

/**
 * 构造树型结构数据
 * @param {*} data 数据源
 * @param {*} id id字段 默认 'id'
 * @param {*} parentId 父节点字段 默认 'parentId'
 * @param {*} children 孩子节点字段 默认 'children'
 */
export function handleTree(data, id, parentId, children) {
    let config = {
        id: id || "id",
        parentId: parentId || "parentId",
        childrenList: children || "children",
    };
    // console.log(parentId, "parentId")
    // console.log(config, "config")
    var childrenListMap = {};
    var nodeIds = {};
    var tree = [];

    for (let d of data) {
        let parentId = d[config.parentId];
        if (childrenListMap[parentId] == null) {
            childrenListMap[parentId] = [];
        }
        nodeIds[d[config.id]] = d;
        childrenListMap[parentId].push(d);
    }

    for (let d of data) {
        let parentId = d[config.parentId];
        // if (nodeIds[parentId] == null) {
        tree.push(d);
        // }
    }

    for (let t of tree) {
        adaptToChildrenList(t);
    }

    function adaptToChildrenList(o) {
        if (childrenListMap[o[config.id]] !== null) {
            o[config.childrenList] = childrenListMap[o[config.id]];
        }
        if (o[config.childrenList]) {
            for (let c of o[config.childrenList]) {
                adaptToChildrenList(c);
            }
        }
    }
    return tree;
}